package com.rickpan.repository;

import com.rickpan.entity.WorkRecord;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDate;
import java.util.List;

/**
 * 工作记录Repository接口
 */
@Repository
public interface WorkRecordRepository extends JpaRepository<WorkRecord, Long> {

    /**
     * 分页查询用户的工作记录
     * @param userId 用户ID
     * @param pageable 分页参数
     * @return 分页结果
     */
    Page<WorkRecord> findByUserIdOrderByDateDescCreatedAtDesc(Long userId, Pageable pageable);

    /**
     * 根据用户ID和日期查询工作记录
     * @param userId 用户ID
     * @param date 日期
     * @return 工作记录列表
     */
    List<WorkRecord> findByUserIdAndDateOrderByCreatedAtAsc(Long userId, LocalDate date);

    /**
     * 根据用户ID和日期查询工作记录（简化方法名）
     */
    List<WorkRecord> findByUserIdAndDate(Long userId, LocalDate date);

    /**
     * 根据用户ID和日期范围查询工作记录
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @return 工作记录列表
     */
    List<WorkRecord> findByUserIdAndDateBetweenOrderByDateAscCreatedAtAsc(Long userId, LocalDate startDate, LocalDate endDate);

    /**
     * 根据用户ID和日期范围查询工作记录（简化方法名）
     */
    List<WorkRecord> findByUserIdAndDateBetween(Long userId, LocalDate startDate, LocalDate endDate);

    /**
     * 根据用户ID和分类查询工作记录
     * @param userId 用户ID
     * @param categoryId 分类ID
     * @param pageable 分页参数
     * @return 分页结果
     */
    Page<WorkRecord> findByUserIdAndCategoryIdOrderByDateDescCreatedAtDesc(Long userId, Long categoryId, Pageable pageable);

    /**
     * 根据用户ID和状态查询工作记录
     * @param userId 用户ID
     * @param status 状态
     * @param pageable 分页参数
     * @return 分页结果
     */
    Page<WorkRecord> findByUserIdAndStatusOrderByDateDescCreatedAtDesc(Long userId, WorkRecord.Status status, Pageable pageable);

    /**
     * 复合条件查询工作记录
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @param categoryId 分类ID（可选）
     * @param status 状态（可选）
     * @param pageable 分页参数
     * @return 分页结果
     */
    @Query("SELECT wr FROM WorkRecord wr WHERE wr.userId = :userId " +
           "AND (:keyword IS NULL OR :keyword = '' OR " +
           "     LOWER(wr.title) LIKE LOWER(CONCAT('%', :keyword, '%')) OR " +
           "     LOWER(wr.content) LIKE LOWER(CONCAT('%', :keyword, '%'))) " +
           "AND (:categoryId IS NULL OR wr.categoryId = :categoryId) " +
           "AND (:status IS NULL OR wr.status = :status) " +
           "AND (:priority IS NULL OR wr.priority = :priority) " +
           "AND (:startDate IS NULL OR wr.date >= :startDate) " +
           "AND (:endDate IS NULL OR wr.date <= :endDate) " +
           "ORDER BY wr.date DESC, wr.createdAt DESC")
    Page<WorkRecord> findByComplexConditions(@Param("userId") Long userId,
                                           @Param("keyword") String keyword,
                                           @Param("categoryId") Long categoryId,
                                           @Param("status") WorkRecord.Status status,
                                           @Param("priority") WorkRecord.Priority priority,
                                           @Param("startDate") LocalDate startDate,
                                           @Param("endDate") LocalDate endDate,
                                           Pageable pageable);

    /**
     * 统计用户在指定日期范围内的工作记录数量
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @return 记录数量
     */
    long countByUserIdAndDateBetween(Long userId, LocalDate startDate, LocalDate endDate);

    /**
     * 统计用户在指定日期范围内各状态的工作记录数量
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @return 统计结果
     */
    @Query("SELECT wr.status, COUNT(wr) FROM WorkRecord wr WHERE wr.userId = :userId " +
           "AND wr.date BETWEEN :startDate AND :endDate GROUP BY wr.status")
    List<Object[]> countByUserIdAndDateBetweenGroupByStatus(@Param("userId") Long userId,
                                                          @Param("startDate") LocalDate startDate,
                                                          @Param("endDate") LocalDate endDate);

    /**
     * 统计用户在指定日期范围内各分类的工作记录数量
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @return 统计结果
     */
    @Query("SELECT wr.categoryId, COUNT(wr) FROM WorkRecord wr WHERE wr.userId = :userId " +
           "AND wr.date BETWEEN :startDate AND :endDate GROUP BY wr.categoryId")
    List<Object[]> countByUserIdAndDateBetweenGroupByCategory(@Param("userId") Long userId,
                                                            @Param("startDate") LocalDate startDate,
                                                            @Param("endDate") LocalDate endDate);

    /**
     * 统计用户在指定日期范围内的总工作时间
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @return 总工作时间（分钟）
     */
    @Query("SELECT COALESCE(SUM(wr.timeSpent), 0) FROM WorkRecord wr WHERE wr.userId = :userId " +
           "AND wr.date BETWEEN :startDate AND :endDate")
    Long sumTimeSpentByUserIdAndDateBetween(@Param("userId") Long userId,
                                          @Param("startDate") LocalDate startDate,
                                          @Param("endDate") LocalDate endDate);

    /**
     * 查找用户最近的工作记录
     * @param userId 用户ID
     * @param limit 限制数量
     * @return 最近的工作记录
     */
    @Query("SELECT wr FROM WorkRecord wr WHERE wr.userId = :userId " +
           "ORDER BY wr.date DESC, wr.createdAt DESC")
    List<WorkRecord> findRecentByUserId(@Param("userId") Long userId, Pageable pageable);
}
